Skip to content

题目链接 · 灵神原题解(署名来源)

想象有一把长度固定的尺子,左端点在链表头部,右端点在正数第 n 个节点。向右移动尺子,当尺子右端点到达链表末尾时,左端点就在倒数第 n 个节点。

由于需要删除节点,我们需要找倒数第 n 个节点的前一个节点(倒数第 n+1 个节点),这样才能做删除操作。

修改:左端点在链表头部,右端点在正数第 n+1 个节点。向右移动尺子,当尺子右端点到达链表末尾时,左端点就在倒数第 n+1 个节点。

细节:如果 n 等于链表长度呢?没有正数第 n+1 个节点。难道要特判这种情况?

不需要。我们可以在头节点的前面插入一个哨兵节点(dummy node),把它当作链表的头节点,这样就有正数第 n+1 个节点了。换句话说,如果遇到需要删除头节点的题目,添加哨兵节点可以简化代码逻辑,请记住这个技巧。

具体请看视频讲解:【基础算法精讲 08】,制作不易,欢迎点赞关注~

答疑

:前后指针的做法为什么算作「一次遍历」?链表节点不是会被遍历两次吗?这个算法的优点是什么?

:链表节点是会被遍历两次。其实我觉得这题本质上想考察的是,在不知道链表长度的前提下,设计一个算法,在「到达链表末尾的瞬间」就能知道倒数第 n 个节点。这个算法的优点是,当 n 比较小,且链表节点分配具有一定局部性时,前后指针的做法 cache miss 更少(相比一个指针跑两趟的算法)。

:一般在做链表题时,什么时候要写 while node,什么时候要写 while node.next

:如果要遍历到最后一个节点,需要写 while node;如果要遍历到倒数第二个节点,需要写 while node.next

python
class Solution:
    def removeNthFromEnd(self, head: Optional[ListNode], n: int) -> Optional[ListNode]:
        # 由于可能会删除链表头部,用哨兵节点简化代码
        left = right = dummy = ListNode(next=head)
        for _ in range(n):
            right = right.next  # 右指针先向右走 n 步
        while right.next:
            left = left.next
            right = right.next  # 左右指针一起走
        left.next = left.next.next  # 左指针的下一个节点就是倒数第 n 个节点
        return dummy.next
cpp
// C++ 版待补充
cpp
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        // 由于可能会删除链表头部,用哨兵节点简化代码
        ListNode dummy{0, head};
        ListNode* left = &dummy;
        ListNode* right = &dummy;
        while (n--) {
            right = right->next; // 右指针先向右走 n 步
        }
        while (right->next) {
            left = left->next;
            right = right->next; // 左右指针一起走
        }
        // 左指针的下一个节点就是倒数第 n 个节点
        ListNode* nxt = left->next;
        left->next = left->next->next;
        delete nxt;
        return dummy.next;
    }
};

复杂度分析

  • 时间复杂度:O(m),其中 m 是链表的长度。
  • 空间复杂度:O(1)

专题训练

见下面链表题单的「§1.5 前后指针」。

分类题单

如何科学刷题?

  1. 滑动窗口与双指针(定长/不定长/单序列/双序列/三指针/分组循环)
  2. 二分算法(二分答案/最小化最大值/最大化最小值/第K小)
  3. 单调栈(基础/矩形面积/贡献法/最小字典序)
  4. 网格图(DFS/BFS/综合应用)
  5. 位运算(基础/性质/拆位/试填/恒等式/思维)
  6. 图论算法(DFS/BFS/拓扑排序/基环树/最短路/最小生成树/网络流)
  7. 动态规划(入门/背包/划分/状态机/区间/状压/数位/数据结构优化/树形/博弈/概率期望)
  8. 常用数据结构(前缀和/差分/栈/队列/堆/字典树/并查集/树状数组/线段树)
  9. 数学算法(数论/组合/概率期望/博弈/计算几何/随机算法)
  10. 贪心与思维(基本贪心策略/反悔/区间/字典序/数学/思维/脑筋急转弯/构造)
  11. 链表、树与回溯(前后指针/快慢指针/DFS/BFS/直径/LCA)
  12. 字符串(KMP/Z函数/Manacher/字符串哈希/AC自动机/后缀数组/子序列自动机)

我的题解精选(已分类)

欢迎关注 B站@灵茶山艾府

本文整理自灵茶山艾府(endlesscheng)的公开内容,仅供个人学习使用

本站仅供个人学习使用,请勿外传